#pragma once
#include "util.hpp"
#include <vector>
#include <cassert>
namespace logs
{
    // 默认缓冲区大小
    #define DEFAULT_BUFFER_SIZE (1 * 1024 * 1024)
    // 缓冲区扩容阈值，低于阈值扩2倍，大于线性增长
    #define THRESHOLD_BUFFER_SIZE (8 * 1024 * 1024)
    // 大于阈值后的线性增长大小
    #define INCREMENT_BUFFER_SIZE (1 * 1024 * 1024)
    class Buffer
    {
    public:
        Buffer():_buffer(DEFAULT_BUFFER_SIZE), _write_idx(0), _read_idx(0)
        {}
        // 写入数据到缓冲区
        void push(const char *data, size_t len){
            // 静态容量
            // if(len > able_write_size()) return ;
            // 动态扩容
            expend_buffer(len);
            // 拷贝数据到缓冲区
            std::copy(data, data + len, &_buffer[_write_idx]);
            // 向后偏移写位置
            move_write(len);
        }
        // 获取读位置起始地址
        const char* begin(){
            return &_buffer[_read_idx];
        }
        // 可读数据长度
        size_t able_read_size(){
            return (_write_idx - _read_idx);
        }
        // 可写数据长度
        size_t able_write_size(){
            // 动态调整方案不需要，空间不够会进行扩容
            // 这里需要给固定大小的缓冲区提供
            return (_buffer.size() - _write_idx);
        }
        // 移动读指针
        void move_read(size_t len){
            assert(len <= able_read_size());
            _read_idx += len;
        }

        // 重置读写位置，初始化缓冲区
        void reset(){
            _write_idx = 0;
            _read_idx = 0;
        }
        // 将缓冲区进行交换
        void swap(Buffer &buffer){
            _buffer.swap(buffer._buffer);
            std::swap(_read_idx, buffer._read_idx);
            std::swap(_write_idx, buffer._write_idx);
        }
        // 判断缓冲区是否为空
        bool empty(){
            return (_read_idx == _write_idx);
        }

    private:
        // 消息太多对缓冲区进行扩容
        void expend_buffer(size_t len)
        {
            if(len <= able_write_size()) return;
            size_t new_size;
            if(_buffer.size() < THRESHOLD_BUFFER_SIZE)
                new_size = _buffer.size() * 2 + len;
            else
                new_size = _buffer.size() + INCREMENT_BUFFER_SIZE + len;
            _buffer.resize(new_size);
        }
        void move_write(size_t n){
            assert(n <= able_write_size());
            _write_idx += n; 
        }
    private:
        // 不选用string, 数据不一定是字符串，vector有更好的扩展性
        // 采用读位置写位置，尽可能的减少拷贝的次数
        // 直接返回string的话，构造string一次拷贝，接受并输出又需要进行一次拷贝
        std::vector<char> _buffer;
        size_t _read_idx;  // 读位置
        size_t _write_idx; // 写位置
    };
}